import socket


def main():
    # 创建套接字
    tcp_client = socket.socket(socket.AF_INET,socket.SOCK_STREAM)

    # 获取服务器IP和端口
    server_ip = input('请输入服务器IP')
    server_port = int(input('请输入服务器PORT'))
    server_addr = (server_ip,server_port)
    
    # 连接服务器
    tcp_client.connect(server_addr)
    
    # 获取，发送要下载的文件名称
    name = input('请输入要下载的文件名')
    tcp_client.send(name.encode('utf-8'))

    # 接受文件
    recv = tcp_client.recv(1024*1024)
    #  recv=recv.decode('utf-8') 因为下面是'wb'模式，可以写入二进制，所以不用解码

    #保存文件
    if recv :
        with open('[接收]'+name,'wb') as f:
            f.write(recv)

    # 关闭套接字
    tcp_client.close()


if __name__ == '__main__':
   main()
